package com.itcam.project.config;

import org.jetbrains.annotations.NotNull;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * @author : Cammy.Wu
 * Description : 线程池
 */

@Configuration
public class ThreadPoolExecutorConfig {

    @Bean
    public ThreadPoolExecutor threadPoolExecutor() {
        // 创建一个线程工厂
        ThreadFactory threadFactory = new ThreadFactory() {
            // 初始化线程数为1
            private int count = 1;

            @Override
            public Thread newThread(@NotNull Runnable r) {
                // 创建一个新的线程
                Thread thread = new Thread(r);
                thread.setName("线程" + count);
                // 线程数递增
                count++;
                // 返回新线程
                return thread;
            }
        };
        // 创建一个新的线程池，线程池核心大小为2，最大线程数为4
        // 非核心线程空闲实践为100秒，任务队列为阻塞队列，长度为4.使用自定义的线程工厂创建线程
        return new ThreadPoolExecutor(
                2, // 核心线程数
                4, // 最大线程数
                100, // 空闲线程存活时间
                TimeUnit.SECONDS, // 空闲线程存活时间的单位
                new ArrayBlockingQueue<>(4), // 工作队列
                threadFactory // 线程工厂
        );
    }
}
